tensor([[[0.2700, 0.2700],
         [0.2700, 0.2700]],

        [[0.3300, 0.3300],
         [0.3300, 0.3300]],

        [[0.4000, 0.4000],
         [0.4000, 0.4000]]], grad_fn=<RoundBackward1>)